% 图像加噪的自定义函数
function noisyImg = add_noise(img, noiseType)
    if strcmp(noiseType, 'gaussian')
        % 添加高斯噪声，均值为0，方差为0.01
        noisyImg = imnoise(img, 'gaussian', 0, 0.01);  
    elseif strcmp(noiseType, 'salt & pepper')
        % 添加椒盐噪声，噪声密度为0.05
        noisyImg = imnoise(img, 'salt & pepper', 0.05);
    else
        error('不支持的噪声类型');
    end
end
